#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project : POS_api_test_project 
@File    : yamlControl.py
@IDE     : PyCharm 
@Author  : Mario
@Date    : 2022/5/26 18:11 
@Describe:
"""
import os
import yaml


class GetYamlData:

    def __init__(self, file_dir):
        self.file_dir = file_dir

    def get_yaml_data(self) -> dict:
        """
        获取 yaml 中的数据
        :param: fileDir: 文件路径
        :return:
        """
        if os.path.exists(self.file_dir):
            data = open(self.file_dir, 'r', encoding='utf-8')
            try:
                res = yaml.load(data, Loader=yaml.FullLoader)
                return res
            except UnicodeDecodeError:
                raise ValueError("yaml文件编码错误,文件路径:{}".format(self.file_dir))
        else:
            raise FileNotFoundError("文件路径不存在~")
